import UIKit
/*
单图会按照图片等比例显示
多图的图片大小固定
多图如果是4张，会按照 2 * 2 显示
多图其他数量，按照 n * 3 九宫格显示
*/
class StatusPictureView: UICollectionView {
    
    let PictureCellId = "PictureCellId"
    
    let itemWidth = (ScreenWidth - StatusCellMargin * 2 - PictureCellMargin * 2) / 3
      //配图视图的数据源数组
    var imageURLs: [NSURL]? {
        didSet {
            //一旦知道了配图视图的具体的数据之后
            //1.计算配图视图的大小
            let psize = calclatePictureViewSize()
            //2.更新配图视图的大小
            self.snp_updateConstraints { (make) -> Void in
                make.size.equalTo(psize)
            }
            //更新测试label的文案:一定要写到set方法里  数据加载后再给label赋值
            testlabel.text = "\(imageURLs?.count ?? 0)"
            //刷新视图  只要tableView 或者 collectionView的数据源发生了更改 就需要调用 reloadData
            reloadData()

        }
    }
    
    override init(frame: CGRect, collectionViewLayout layout: UICollectionViewLayout) {
        //实例化流水布局
        let flowLayout = UICollectionViewFlowLayout()
        //设置layout
        flowLayout.itemSize = CGSizeMake(itemWidth, itemWidth)
        //设置itemSize
        flowLayout.minimumInteritemSpacing = 0
        flowLayout.minimumLineSpacing = 0
        
        super.init(frame: frame, collectionViewLayout: flowLayout)
        self.registerClass(PictureCell.self, forCellWithReuseIdentifier: PictureCellId)
        //设置数据源
        self.dataSource = self
        //设置背景色
        self.backgroundColor = UIColor.whiteColor()
        //添加测试label
        addSubview(testlabel)
                testlabel.snp_makeConstraints { (make) -> Void in
            make.center.equalTo(self.center)
        }
    }

    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

    private func calclatePictureViewSize() -> CGSize {
        //1.获取图片数量
        let imagecount = imageURLs?.count ?? 0
        if imagecount == 0 {
            return CGSizeZero
        }
        
        let layout = self.collectionViewLayout as! UICollectionViewFlowLayout

        //如果是单张图片
        if imagecount == 1 {
            //返回和图片大小一致的size
            //TODO: 后期完善
//            let imageSize = CGSize(width: 150, height: 250)
            let imageSize = CGSize(width: 150, height: 200)
            
            layout.itemSize = imageSize
            
            return imageSize
        }
        
        layout.itemSize = CGSize(width: itemWidth, height: itemWidth)
        
        if imagecount == 4 {
            //配图视图大小的计算
            let w = itemWidth * 2 + PictureCellMargin
            return CGSize(width: w, height: w)
        }
        
        //程序走到这个地方 就是其他多张图片显示的情况
        /*
        1,2,3   -> 1
        4,5,6   -> 2
        7,8,9   -> 3
        */
        //计算行数
        let row = CGFloat((imagecount - 1) / 3 + 1)
        let w = itemWidth * 3 + PictureCellMargin * 2
        let h = itemWidth * row + PictureCellMargin * (row - 1)
        return CGSizeMake(w, h)

    }
    private lazy var testlabel: UILabel = UILabel(text: "", fontSize: 40, textColor: UIColor.redColor())
}

extension StatusPictureView: UICollectionViewDataSource {
    func collectionView(collectionView: UICollectionView, numberOfItemsInSection section: Int) -> Int {
        return imageURLs?.count ?? 0
    }
    
    func collectionView(collectionView: UICollectionView, cellForItemAtIndexPath indexPath: NSIndexPath) -> UICollectionViewCell {
        let cell = collectionView.dequeueReusableCellWithReuseIdentifier(PictureCellId, forIndexPath: indexPath) as! PictureCell
        cell.imageURL = imageURLs![indexPath.item]
        //设置调试颜色(奇偶不同)
        cell.backgroundColor = indexPath.item % 2 == 0 ? UIColor.redColor() : UIColor.blueColor()
        
        return cell
    }
    
}
//类之间关系很密切 并且没有太多复杂的逻辑 是可以写在同一个文件中
private class PictureCell: UICollectionViewCell {
    //外部设置的图片url属性
    var imageURL: NSURL? {
        didSet {
            iconView.sd_setImageWithURL(imageURL)
        }
    }
    //重写构造方法
    override init(frame: CGRect) {
        super.init(frame: frame)
        self.backgroundColor = UIColor(white: 0.95, alpha: 1)
        setupUI()
    }
    
    required init?(coder aDecoder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
    private func setupUI() {
        contentView.addSubview(iconView)
        //设置约束
        iconView.snp_makeConstraints { (make) -> Void in
            make.edges.equalTo(contentView.snp_edges)
        }
        
    }
    //懒加载视图
    private lazy var iconView: UIImageView = {
        let iv = UIImageView()
        //UIImageView默认的显示方式  ScaleToFill: 缩放全部视图进行填充
        //显得很山寨
        //        iv.contentMode = .ScaleToFill
        // Aspect: 部分  缩放部分 进行填充  视图的比例不会失真 但是 图片会被裁剪
        iv.contentMode = .ScaleAspectFill
        //缩放图片  部分图片进行适应的显示  会导致 imageView可能会留白  一般也不推荐使用这样方式来显示图片
        //        iv.contentMode = .ScaleAspectFit
        //手动设置裁剪
        iv.clipsToBounds = true
        return iv
    }()

}

